package cool.webstudy.async.configuration.mybatisplus;
/**
 *
 * @author think
 * @date 2024/8/1 23:08
 * @description:  动态表名帮助类，将表名添加至线程变量中，使得表名能够在拦截器中读取到，从而实现动态表名
 */
public class DynamicTableNameHelper {
    /**
     * 线程变量中存取（表名）
     */
    private static final ThreadLocal<String> TABLE_NAME = new ThreadLocal<>();

    /**
     * 设置请求参数
     */
    public static void setTableName(String tableName) {
        TABLE_NAME.set(tableName);
    }

    /**
     * 获取请求参数
     */
    public static String getTableName() {
        return TABLE_NAME.get();
    }

    /**
     * 移除线程变量
     */
    public static void remove() {
        TABLE_NAME.remove();
    }
}
